#!/bin/bash
# Users can modify the following paths
DATASET_DIR="/backup/data/Audioset/zelin"
WORKSPACE="/backup/data/Audioset/zelin/workspaces/audioset_tagging"

# ============ Train & Inference ============
MODEL_TYPE="Cnn14Bert"

# # Plot statistics
# python3 -u utils/plot_statistics_bimodal.py plot --dataset_dir=$DATASET_DIR --workspace=$WORKSPACE --select=1_bal_bert --window_size 512

# python3 -u utils/plot_statistics_bimodal.py plot --dataset_dir=$DATASET_DIR --workspace=$WORKSPACE --select=1_bal_bert --window_size 1024

python3 -u utils/plot_statistics_bimodal.py plot --dataset_dir=$DATASET_DIR --workspace=$WORKSPACE --select=1_bal_bert --window_size 1024 --bert_model_type prajjwal1/bert-medium --max_text_nums 128